How can i make this Python code less repetitive